package com.heima.behavior.service.impl;

import com.heima.behavior.service.ApBehaviorEntryService;
import com.heima.behavior.service.ApCollectionBehaviorService;
import com.heima.common.exception.CustException;
import com.heima.model.behavior.dtos.CollectionBehaviorDTO;
import com.heima.model.behavior.pojos.ApBehaviorEntry;
import com.heima.model.behavior.pojos.ApCollection;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.threadlocal.AppThreadLocalUtils;
import com.heima.model.user.pojos.ApUser;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Service;

import java.util.Date;

@Service
public class ApCollectionBehaviorServiceImpl implements ApCollectionBehaviorService {
    @Autowired
    MongoTemplate mongoTemplate;
    @Autowired
    ApBehaviorEntryService apBehaviorEntryService;
    @Override
    public ResponseResult collectBehavior(CollectionBehaviorDTO dto) {
        // 1. 校验参数 需要登录 文章id不能null  操作类型 0 或 1
        ApUser user = AppThreadLocalUtils.getUser();
        if (user == null) {
            CustException.cust(AppHttpCodeEnum.NEED_LOGIN);
        }
        // 2. 根据userId查询行为实体数据
        ApBehaviorEntry behaviorEntry = apBehaviorEntryService.findByUserIdOrEquipmentId(user.getId(), dto.getEquipmentId());
        if (behaviorEntry == null) {
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST,"获取行为实体信息失败");
        }
        // 3. 如果是收藏操作 判断是否已经收藏
        Query query = Query.query(Criteria.where("entryId").is(behaviorEntry.getId()).and("articleId").is(dto.getArticleId()));
        ApCollection collection = mongoTemplate.findOne(query, ApCollection.class);
        if (dto.getOperation().intValue() == 0) {
            if (collection != null) {
                CustException.cust(AppHttpCodeEnum.DATA_EXIST,"该文章已被收藏");
            }
            // 4. 如果未收藏 新增收藏行为
            collection = new ApCollection();
            collection.setEntryId(behaviorEntry.getId());
            collection.setArticleId(dto.getArticleId());
            collection.setType(dto.getType());
            collection.setCollectionTime(new Date());
            mongoTemplate.save(collection);
        } else {
            // 5. 如果是取消收藏操作  删除收藏行为
            mongoTemplate.remove(query, ApCollection.class);
        }
        return ResponseResult.okResult();
    }
}
